import sys
import os


# Add the project root to the Python path
project_dir_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
sys.path.append(project_dir_path)
print(project_dir_path)

from omics_bert_datasets_2_embeddings.write_embeddings import EmbeddingWriter

if __name__ == '__main__':
    texts = ['This is a test sentence.', 'This is another test sentence.']
    with EmbeddingWriter('test_embeddings',tensor_parallel_size=1) as embedding_writer:
        # for text in texts:
        #     embedding_writer.embedding_and_write(text)
        embedding_writer.embedding_and_write_batch(texts)
